package work.thread_eg.concurrent.lock;

import java.util.Random;
import java.util.concurrent.Phaser;
import java.util.concurrent.TimeUnit;

/**
 *  同步组件Phaser 用来解决控制多个线程分阶段共同完成任务的情景问题。它与CountDownLatch和CyclicBarrier类似
 *  但是可以动态控制parties
 *
 *  多个阶段 每个阶段是多个线程 相当于每个阶段是一个CyclicBarrier 类似升级版
 */
public class PhaserTest01 {

    static Random r = new Random();
    static MarriagePhaser phaser = new MarriagePhaser();

    public static void main(String[] args) {
        // 一共七个线程  phaser对象是一个
        // phaser要注册是7个线程数
        // 保证每个阶段所有线程都执行完了 再执行下个阶段
        phaser.bulkRegister(7);

        // 构建线程
        for(int i=0 ; i <5 ;i++) {
            new Thread(new Person("p"+i)).start();
        }
        new Thread(new Person("新郎")).start();
        new Thread(new Person("新娘")).start();

    }

    static void sleep(int mill) {
        try {
            TimeUnit.MILLISECONDS.sleep(mill);
        } catch (Exception ex) {
            ex.printStackTrace();
        }
    }

    static class Person implements Runnable{
        String name;

        public Person(String name) {
            this.name = name;
        }

        public void arrive() {
            System.out.println(name + "到达现场");
            sleep(r.nextInt(1000));
            // 七个线程执行到arriveAndAwaitAdvance() 方法之后 都会等待在这里
            // 最后一个线程执行到这里之后会执行onAdvance方法 这时候phase是0 表示是阶段一
            // 并且所有都会往下执行了
            phaser.arriveAndAwaitAdvance();

        }

        public void eat() {
            System.out.println(name + "吃完");
            sleep(r.nextInt(1000));
            phaser.arriveAndAwaitAdvance();
        }

        public void leave() {
            System.out.println(name + "离开");
            sleep(r.nextInt(1000));
            phaser.arriveAndAwaitAdvance();
        }

        public void hug() {
            if("新郎".equals(name) || "新娘".equals(name)) {
                sleep(r.nextInt(1000));
                System.out.println(name + "洞房");
                phaser.arriveAndAwaitAdvance();
            }else {
                // 到达之后解除注册 所以 onAdvance的第二个参数才是2 其他线程解除注册了
                phaser.arriveAndDeregister();
            }

        }

        @Override
        public void run() {
            // 一共七个线程 四个阶段
            arrive();
            eat();
            leave();
            hug();
        }
    }


    static class MarriagePhaser extends Phaser {
        /**
         * return false 表示所有个阶段结束了 不往下继续了
         * @param phase                 阶段的序号
         * @param registeredParties     对应阶段的等待的线程数
         * @return
         */
        @Override
        protected boolean onAdvance(int phase, int registeredParties) {
            switch (phase) {
                case 0:
                    System.out.println("所有人到齐了!" + registeredParties);
                    System.out.println();
                    return false;
                case 1:
                    System.out.println("所有人吃完了!" + registeredParties);
                    System.out.println();
                    return false;
                case 2:
                    System.out.println("所有人离开了!" + registeredParties);
                    System.out.println();
                    return false;
                case 3:
                    System.out.println("婚礼结束!" + registeredParties);
                    return true;
                default:
                    return true;
            }

        }

    }
}
